package test;

import ai.djl.ndarray.NDArray;
import ai.djl.ndarray.NDManager;
import ai.djl.ndarray.index.NDIndex;
import ai.djl.ndarray.types.DataType;
import ai.djl.ndarray.types.Shape;

public class NdArrayDemo {

    public static void main(String[] args) {
        NDManager ndManager = NDManager.newBaseManager();
        NDArray ndArray = ndManager.arange(12).reshape(3,-1).transpose();
        System.out.println(ndArray);

        ndArray.set(new NDIndex("1,1"), array->array.add(1));

        System.out.println(ndArray);
    }
}
